# gomoku_ai/config.py

# 游戏配置
BOARD_SIZE = 15

# 训练配置
NUM_ITERATIONS = 30
NUM_EPISODES_PER_ITER = 20
BATCH_SIZE = 512
TRAIN_EPOCHS_PER_ITER = 10
LEARNING_RATE = 0.001
MCTS_SIMULATIONS = 50

# 文件路径配置
CHECKPOINT_DIR = "./checkpoints"
MODEL_FILENAME_TEMPLATE = "gomoku_model_iter_{iteration}.pth"

# 界面配置
VISUAL_TRAINING = True  # 默认训练时是否显示图形界面

# --- 经典AI配置 ---
# 经典AI的搜索深度，越大越强，但训练越慢
# 1: 简单 (随机下棋)
# 2: 中等 (会攻防)
# 3: 困难 (会评估位置)
# 4: 专家级 (更深的评估)
# 5: 大师级 (非常深的评估)
CLASSIC_AI_DEPTH = 4

# --- 自我对弈配置 ---
# 设置为 True，神经网络将与自己的历史版本对战（AlphaZero风格）
# 设置为 False，神经网络将与经典AI对战（更稳定，适合初期训练）
USE_SELF_PLAY = False
